machine learning

Waste detection

An YOLO model specifically trained to detect waste in images. We collected a custom dataset of waste, classified and annotated it, and trained the model using YOLOv8 on Keras. The model is able to detect the position and class of waste in images. It can separate the following classes of waste: